from car import Car

class Battery():
    def __init__(self,battery_size=70):
        self.battery_size = battery_size

    def describe_battery(self):
        print("This car has " + str(self.battery_size) + "-KWH")

    def get_range(self):
        if self.battery_size == 70:
            range = 250
        elif self.battery_size == 80:
            range = 300
        else:
            range = 200
        message = "This car can go " + str(range)
        message += " miles on a full charge. "
        print(message)

class ElectricCar(Car):
    def __init__(self,make,model,year):
        super().__init__(make,model,year)
        self.battery = Battery()